#include <my_header.h>
// 服务端不关闭 客户端可多次重连
/* Usage: ./11_server_epoll_断线重连 */
int main(void){                                  
    // 服务器的ip地址和端口
    char *ip = "192.168.229.128";
    char *port = "8080";

    // 创建socket对象
    int sofd = socket(AF_INET, SOCK_STREAM, 0);
    ERROR_CHECK(sofd, -1, "socket");

    // 超时重连
    int reuse = 1;
    int res_set = setsockopt(sofd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse));
    ERROR_CHECK(res_set, -1, "setsockopt");

    // 创建struct sockaddr 类型
    struct sockaddr_in server_addr;
    server_addr.sin_family  = AF_INET;
    server_addr.sin_port = htons(atoi(port));
    server_addr.sin_addr.s_addr = inet_addr(ip);

    // 绑定端口和ip
    int res_bind = bind(sofd, (struct sockaddr *)&server_addr, sizeof(server_addr));
    ERROR_CHECK(res_bind, -1, "bind");

    // 监听
    int res_lis = listen(sofd, 10 );
    ERROR_CHECK(res_lis, -1, "listen");

    // 获取全连接队列描述符
    int net_fd ;

    // 创建epoll对象
    int epoll_fd = epoll_create(1);
    ERROR_CHECK(epoll_fd, -1, "epoll_create");
    // 设置监听集合
    struct epoll_event event;
    event.events = EPOLLIN;
    event.data.fd = sofd;
    int res_epoctl = epoll_ctl(epoll_fd, EPOLL_CTL_ADD, sofd, &event);
    ERROR_CHECK(res_epoctl, -1, "epoll_ctl");
    while (1) {  
        // 创建一个epoll_event数组，用于接收发生的事件  
        struct epoll_event list[10];  

        // 等待epoll事件发生，-1表示无限等待  
        int epoll_num = epoll_wait(epoll_fd, list, 10, -1);  

        // 遍历所有发生的事件  
        for (int i = 0; i < epoll_num; i++) {  
            int fd = list[i].data.fd; // 获取发生事件的文件描述符  

            // 如果文件描述符是sofd (服务器端套接字)  
            if (fd == sofd) {  
                // 接受新的连接，返回新的net_fd (客户端套接字)  
                net_fd = accept(sofd, NULL, NULL);  

                // 移除sofd的监听，因为它只用于接受连接  
                epoll_ctl(epoll_fd, EPOLL_CTL_DEL, sofd, NULL);  

                // 添加stdin的监听  
                event.data.fd = STDIN_FILENO;   // 设定event的文件描述符为标准输入  
                epoll_ctl(epoll_fd, EPOLL_CTL_ADD, STDIN_FILENO, &event);  

                // 添加net_fd的监听  
                event.data.fd = net_fd;          // 设定event的文件描述符为新建立的net_fd  
                epoll_ctl(epoll_fd, EPOLL_CTL_ADD, net_fd, &event);  

                continue; // 继续下次循环，等待下一个事件  
            }  

            // 如果文件描述符是STDIN_FILENO (标准输入)  
            if (fd == STDIN_FILENO) {  
                char buf[60] = {0};            // 创建缓冲区以存储输入  
                                               // 从标准输入读取数据  
                read(STDIN_FILENO, buf, sizeof(buf));  
                // 将读取到的数据发送到net_fd (客户端)  
                send(net_fd, buf, sizeof(buf), 0);  
            }  

            // 如果文件描述符是来自net_fd (客户端)  
            if (fd == net_fd) {  
                char buf[60] = {0};            // 创建缓冲区以存储接收的数据  
                ssize_t ret = recv(net_fd, buf, sizeof(buf), 0); // 从net_fd接收数据  

                // 如果ret为0，表示客户端关闭了连接  
                if (ret == 0) {  
                    // 移除标准输入和net_fd的监听  
                    epoll_ctl(epoll_fd, EPOLL_CTL_DEL, STDIN_FILENO, NULL);  
                    epoll_ctl(epoll_fd, EPOLL_CTL_DEL, net_fd, NULL);  

                    close(net_fd); // 关闭net_fd (客户端套接字)  

                    // 重新加入sofd的监听，以接受新的连接  
                    event.data.fd = sofd;  
                    epoll_ctl(epoll_fd, EPOLL_CTL_ADD, sofd, &event);  

                    continue; // 继续下次循环  
                }  

                // 打印接收到的数据  
                printf("client message: %s \n", buf);  
            }  
        }  
    }  

    // 关闭net_fd和sofd  
    close(net_fd);  
    close(sofd);  

    return 0;
}
